import json

import utils.env_conf
from utils.OpenAiService import getParam, getFun
from utils.StaticObj import SYSTEM_PROMPT, ORG_CONF, FUNCTIONS, FUN_DICT

un = 'debug'
userRole = 'user2'
systemPrompt = SYSTEM_PROMPT.format('2023年8月3日', ORG_CONF[userRole])
print(systemPrompt)

query = '我想查看2021年12月16号A项目1标拌合站的各种碎石进场数量是多少'

paramsMsg = getParam(un, query, FUNCTIONS, sysStr=systemPrompt)
if paramsMsg.function_call:
    functionName = paramsMsg.function_call['name']
    functionArgs = json.loads(paramsMsg.function_call['arguments'])
    data = FUN_DICT[functionName](functionArgs)
    print(data)
    answer = getFun(un, data, functionName, sysStr=systemPrompt)
    print(answer.content)

